home *** CD-ROM | disk | FTP | other *** search
/ Chip 2000 October / CHIP Turkiye Ekim 2000.iso / prog / naps / 04 / setup.exe / Gnucleus / MainFrm.cpp < prev    next >
Encoding:
C/C++ Source or Header  |  2000-07-15  |  23.2 KB  |  907 lines
  1. /********************************************************************************
  2.  
  3.     Gnucleus - A node application for the Gnutella network
  4.     Copyright (C) 2000 John Marshall
  5.  
  6.     This program is free software; you can redistribute it and/or modify
  7.     it under the terms of the GNU General Public License as published by
  8.     the Free Software Foundation; either version 2 of the License.
  9.  
  10.     This program is distributed in the hope that it will be useful,
  11.     but WITHOUT ANY WARRANTY; without even the implied warranty of
  12.     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
  13.     GNU General Public License for more details.
  14.  
  15.     You should have received a copy of the GNU General Public License
  16.     along with this program; if not, write to the Free Software
  17.     Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  18.  
  19.     For support, questions, comments, etc...
  20.     E-Mail: 
  21.         swabby@c0re.net
  22.     
  23.     Address:
  24.         21 Cadogan Way
  25.         Nashua, NH, USA 03062
  26.  
  27. ********************************************************************************/
  28.  
  29. // MainFrm.cpp : implementation of the CMainFrame class
  30. //
  31.  
  32. #include "stdafx.h"
  33. #include "Gnucleus.h"
  34. #include "GnucleusDoc.h"
  35.  
  36. #include "ViewConnect.h"
  37. #include "MainFrm.h"
  38.  
  39. #include "GnuHash.h"
  40. #include "GnuControl.h"
  41.  
  42. #define    WM_ICON_NOTIFY            WM_USER+10
  43.  
  44. #ifdef _DEBUG
  45. #define new DEBUG_NEW
  46. #undef THIS_FILE
  47. static char THIS_FILE[] = __FILE__;
  48. #endif
  49.  
  50. /////////////////////////////////////////////////////////////////////////////
  51. // CMainFrame
  52.  
  53. IMPLEMENT_DYNAMIC(CMainFrame, CMDIFrameWnd)
  54.  
  55. BEGIN_MESSAGE_MAP(CMainFrame, CMDIFrameWnd)
  56.     //{{AFX_MSG_MAP(CMainFrame)
  57.     ON_WM_CREATE()
  58.     ON_WM_TIMER()
  59.     ON_COMMAND(ID_OPEN, OnOpenFromTray)
  60.     ON_WM_SHOWWINDOW()
  61.     ON_WM_NCACTIVATE()
  62.     ON_WM_NCLBUTTONUP()
  63.     ON_WM_NCLBUTTONDOWN()
  64.     ON_WM_NCMOUSEMOVE()
  65.     ON_WM_NCPAINT()
  66.     ON_MESSAGE(WM_ICON_NOTIFY, OnTrayNotification)
  67.     ON_WM_PAINT()
  68.     //}}AFX_MSG_MAP
  69. END_MESSAGE_MAP()
  70.  
  71. static UINT indicators[] =
  72. {
  73.     ID_SEPARATOR,     // status line indicator
  74.     ID_INDICATOR_RECV,
  75.     ID_INDICATOR_SEND
  76. };
  77.  
  78. /////////////////////////////////////////////////////////////////////////////
  79. // CMainFrame construction/destruction
  80.  
  81. CMainFrame::CMainFrame()
  82. {
  83.     Doc = (CGnucleusDoc *) ((CGnucleusApp *) AfxGetApp())->MasterDoc;
  84.    m_bUseTrayIcon = TRUE;
  85.     Strike = 0;
  86.    InitializeTrayIcon();
  87. }
  88.  
  89. CMainFrame::~CMainFrame()
  90. {
  91.     RemoveIcon();
  92.     m_wndTrayIcon.DestroyWindow();
  93. }
  94.  
  95. int CMainFrame::OnCreate(LPCREATESTRUCT lpCreateStruct)
  96. {
  97.     if (CMDIFrameWnd::OnCreate(lpCreateStruct) == -1)
  98.         return -1;
  99.     
  100.     if (!m_wndToolBar.CreateEx(this) ||
  101.         !m_wndToolBar.LoadToolBar(IDR_MAINFRAME))
  102.     {
  103.         TRACE0("Failed to create toolbar\n");
  104.         return -1;      // fail to create
  105.     }
  106.     if (!m_wndDlgBar.Create(this, IDR_MAINFRAME, 
  107.         CBRS_ALIGN_TOP, AFX_IDW_DIALOGBAR))
  108.     {
  109.         TRACE0("Failed to create dialogbar\n");
  110.         return -1;        // fail to create
  111.     }
  112.     else
  113.     {
  114.         CComboBox* cbptr = ((CComboBox *) m_wndDlgBar.GetDlgItem(IDC_COMBO_SPEED));
  115.  
  116.         cbptr->InsertString(0, "T3 (or Greater)");
  117.         cbptr->InsertString(0, "T1");
  118.         cbptr->InsertString(0, "DSL");
  119.         cbptr->InsertString(0, "Cable");
  120.         cbptr->InsertString(0, "ISDN-128K");
  121.         cbptr->InsertString(0, "ISDN-56K");
  122.         cbptr->InsertString(0, "56K Modem");
  123.         cbptr->InsertString(0, "33.6 Modem");
  124.         cbptr->InsertString(0, "28.8 Modem");
  125.         cbptr->InsertString(0, "14.4 Modem");
  126.         cbptr->InsertString(0, "");
  127.     }
  128.  
  129.     if (!m_wndReBar.Create(this) ||
  130.         !m_wndReBar.AddBar(&m_wndToolBar) ||
  131.         !m_wndReBar.AddBar(&m_wndDlgBar))
  132.     {
  133.         TRACE0("Failed to create rebar\n");
  134.         return -1;      // fail to create
  135.     }
  136.  
  137.     if (!m_wndStatusBar.Create(this) ||
  138.         !m_wndStatusBar.SetIndicators(indicators,
  139.           sizeof(indicators)/sizeof(UINT)))
  140.     {
  141.         TRACE0("Failed to create status bar\n");
  142.         return -1;      // fail to create
  143.     }
  144.     else
  145.     {
  146.         m_wndStatusBar.SetPaneInfo(1, ID_INDICATOR_RECV, SBPS_NORMAL, 140);
  147.         m_wndStatusBar.SetPaneInfo(2, ID_INDICATOR_SEND, SBPS_NORMAL, 125);
  148.     }
  149.  
  150.     if (!CreateTrayIcon(this, WM_ICON_NOTIFY, _T("Gnucleus"), AfxGetApp()->LoadIcon(IDR_MAINFRAME),
  151.                             IDR_TRAY))
  152.     {
  153.         TRACE0("Failed to create system tray icon.\n");
  154.         return -1;      // fail to create
  155.     }
  156.     else
  157.     {
  158.         HideIcon();
  159.       if (m_bOwnerCreated) 
  160.       {
  161.          ShowTaskBarButton(TRUE);
  162.       }
  163.     }
  164.  
  165.    // TODO: Remove this if you don't want tool tips
  166.     m_wndToolBar.SetBarStyle(m_wndToolBar.GetBarStyle() |
  167.         CBRS_TOOLTIPS | CBRS_FLYBY);
  168.  
  169.     UpdateTimer = SetTimer(1, 1000, 0);    
  170.  
  171.     return 0;
  172. }
  173.  
  174. void CMainFrame::InitializeTrayIcon()
  175. {
  176.     memset(&m_tnd, 0, sizeof(m_tnd));
  177.     m_bEnabled   = FALSE;
  178.     m_bHidden    = FALSE;
  179.     m_hSavedIcon = NULL;
  180.     m_DefaultMenuItemID = 0;
  181.     m_DefaultMenuItemByPos = TRUE;
  182. }
  183.  
  184. BOOL CMainFrame::CreateTrayIcon(CWnd* pParent, UINT uCallbackMessage, LPCTSTR szToolTip, 
  185.                          HICON icon, UINT uID)
  186. {
  187.     // this is only for Windows 95 (or higher)
  188.     VERIFY(m_bEnabled = ( GetVersion() & 0xff ) >= 4);
  189.     if (!m_bEnabled) return FALSE;
  190.  
  191.     // Make sure Notification window is valid (not needed - CJM)
  192.     // VERIFY(m_bEnabled = (pParent && ::IsWindow(pParent->GetSafeHwnd())));
  193.     // if (!m_bEnabled) return FALSE;
  194.     
  195.     // Make sure we avoid conflict with other messages
  196.     ASSERT(uCallbackMessage >= WM_USER);
  197.  
  198.     // Tray only supports tooltip text up to 64 characters
  199.     ASSERT(_tcslen(szToolTip) <= 64);
  200.  
  201.     // Create an invisible window
  202.     m_wndTrayIcon.CreateEx(0, AfxRegisterWndClass(0), _T(""), WS_POPUP, 0,0,10,10, NULL, 0);
  203.  
  204.     // load up the NOTIFYICONDATA structure
  205.     m_tnd.cbSize = sizeof(NOTIFYICONDATA);
  206.     m_tnd.hWnd   = pParent->GetSafeHwnd()? pParent->GetSafeHwnd() : m_wndTrayIcon.GetSafeHwnd();
  207.     m_tnd.uID    = uID;
  208.     m_tnd.hIcon  = icon;
  209.     m_tnd.uFlags = NIF_MESSAGE | NIF_ICON | NIF_TIP;
  210.     m_tnd.uCallbackMessage = uCallbackMessage;
  211.     _tcscpy(m_tnd.szTip, szToolTip);
  212.  
  213.     // Set the tray icon
  214.     VERIFY(m_bEnabled = Shell_NotifyIcon(NIM_ADD, &m_tnd));
  215.     return m_bEnabled;
  216. }
  217.  
  218. BOOL CMainFrame::PreCreateWindow(CREATESTRUCT& cs) 
  219. { 
  220.    if (!CMDIFrameWnd::PreCreateWindow(cs)) return FALSE; 
  221.    if (!m_bOwnerCreated) // "dialog-is-created" flag 
  222.    { 
  223.      // m_MainFrameOwner - CDialog object mapped to the resource dialog template 
  224.      m_bOwnerCreated = m_MainFrameOwner.Create(IDD_OWNERFRAME); 
  225.      if (m_bOwnerCreated) m_MainFrameOwner.ShowWindow(SW_HIDE); 
  226.    }; 
  227.  
  228.    // set the dialog as a parent of CMainFrame window 
  229.    if (m_bOwnerCreated) 
  230.    {
  231.       cs.hwndParent = m_MainFrameOwner.GetSafeHwnd(); 
  232.    }
  233.  
  234.    return TRUE; 
  235. } 
  236.  
  237. BOOL CMainFrame::ShowTaskBarButton(BOOL bVisible) 
  238. { 
  239.     if (!m_bOwnerCreated) return FALSE; 
  240.  
  241. //    ShowWindow(SW_HIDE);    // seems to cause various problems,
  242.                             // ranging from flashing windows to the title bar disappearing
  243.  
  244.     if (bVisible) 
  245.         ModifyStyleEx(0, WS_EX_APPWINDOW); 
  246.     else 
  247.         ModifyStyleEx(WS_EX_APPWINDOW, 0);
  248.  
  249. //    ShowWindow(SW_SHOW);
  250.  
  251.     return TRUE; 
  252. } 
  253. /////////////////////////////////////////////////////////////////////////////
  254. // CMainFrame diagnostics
  255.  
  256. #ifdef _DEBUG
  257. void CMainFrame::AssertValid() const
  258. {
  259.     CMDIFrameWnd::AssertValid();
  260. }
  261.  
  262. void CMainFrame::Dump(CDumpContext& dc) const
  263. {
  264.     CMDIFrameWnd::Dump(dc);
  265. }
  266.  
  267. #endif //_DEBUG
  268.  
  269. /////////////////////////////////////////////////////////////////////////////
  270. // CMainFrame message handlers
  271.  
  272. CDialogBar* CMainFrame::GetDialogBar()
  273. {
  274.     return &m_wndDlgBar;
  275. }
  276.  
  277. void CMainFrame::OnTimer(UINT nIDEvent) 
  278. {    
  279.  
  280.     if(nIDEvent == UpdateTimer)
  281.     {
  282.         int  decimal, sign;
  283.         float BytesIn = (float) Doc->GnuComm->BytesIn,
  284.               BytesOut = (float) Doc->GnuComm->BytesOut,
  285.               ByteLimit = (float) Doc->m_LimitTotal,
  286.               Kilo = 1024;
  287.  
  288.         // call thread flow control OnTimer only once a second
  289.         Doc->GnuComm->OnTimer();
  290.  
  291.         
  292.         // Reset
  293.         Doc->GnuComm->BytesIn  = 0;
  294.         Doc->GnuComm->BytesOut = 0;
  295.  
  296.         // The the estimated speed for gnucleus
  297.         Doc->m_EstSpeed = GetEstSpeed(BytesIn * 8 / 1024);
  298.  
  299.         // Receive status bar update
  300.         float Old_RecvSpeed = atof(m_wndStatusBar.GetPaneText(1));
  301. //        CString Speed = _fcvt( ((BytesIn / Kilo) + Old_RecvSpeed) / 2, 2, &decimal, &sign );
  302.         CString Speed = _fcvt( Doc->GnuComm->m_dwBytesPerSecIn / Kilo, 2, &decimal, &sign );
  303.         Speed.Insert(decimal, ".");
  304.  
  305.         CString Title = "Receive rate:   ";
  306.         Title += Speed;
  307.         Title += " KB/s";
  308.         m_wndStatusBar.SetPaneText(1, Title);
  309.  
  310.         // Send status bar update
  311.         float Old_SendSpeed = atof(m_wndStatusBar.GetPaneText(2));
  312. //        Speed = _fcvt( ((BytesOut / Kilo) + Old_SendSpeed) / 2, 2, &decimal, &sign );
  313.         Speed = _fcvt( Doc->GnuComm->m_dwBytesPerSecOut / Kilo, 2, &decimal, &sign );
  314.         Speed.Insert(decimal, ".");
  315.  
  316.         Title = "Send rate:   ";
  317.         Title += Speed;
  318.         Title += " KB/s";
  319.         m_wndStatusBar.SetPaneText(2, Title);
  320.  
  321. /*
  322.         // Drop if limit hit
  323.         float AverageSpeed = (((BytesOut + BytesIn) / Kilo) + (Old_RecvSpeed + Old_SendSpeed)) / 2;
  324.  
  325.         if(ByteLimit != 0)
  326.             if(ByteLimit < AverageSpeed)
  327.                 if(AverageSpeed - ByteLimit > 1)
  328.                     Strike++;
  329.                 else
  330.                     Strike = 0;
  331. */
  332.  
  333.         // If limit is hit 3 times in a row, drop the connection
  334.         if(Strike > 10)
  335.         {    
  336.             Strike = 0;
  337.  
  338.             Doc->m_ConnectNum--;
  339.  
  340.             if( Doc->m_ConnectNum == 0)
  341.                 AfxMessageBox("Not enough bandwidth to support Gnucleus.");
  342.  
  343.             ((CViewConnect * ) ((CGnucleusApp *) AfxGetApp())->ConnectFrame->GetActiveView())->RemoveFirstNode();
  344.         }
  345.     }
  346.  
  347.     CMDIFrameWnd::OnTimer(nIDEvent);
  348. }
  349.  
  350. DWORD CMainFrame::GetEstSpeed(DWORD CurrentSpeed)
  351. {
  352.     if(CurrentSpeed <= Doc->m_EstSpeed)
  353.         return Doc->m_EstSpeed;
  354.     else
  355.     {
  356.         if(CurrentSpeed > 45000)
  357.             return 45000;
  358.         if(CurrentSpeed > 1500)
  359.             return 1500;
  360.         if(CurrentSpeed > 768)
  361.             return 768;
  362.         if(CurrentSpeed > 384)
  363.             return 384;
  364.         if(CurrentSpeed > 128)
  365.             return 128;
  366.         if(CurrentSpeed > 56)
  367.             return 56;
  368.         if(CurrentSpeed > 53)
  369.             return 53;
  370.         if(CurrentSpeed > 29)
  371.             return 29;
  372.         if(CurrentSpeed > 14)
  373.             return 14;
  374.     }
  375.     
  376.     return 0;
  377. }
  378.  
  379. BOOL CMainFrame::OnCommand(WPARAM wParam, LPARAM lParam) 
  380. {
  381.     // Entering return in they keyword box submits a search
  382.     if(wParam == 1 && lParam == 0)
  383.     {
  384.         ((CGnucleusApp *) AfxGetApp())->OnButtonSearch();
  385.     }
  386.     
  387.     return CMDIFrameWnd::OnCommand(wParam, lParam);
  388. }
  389.  
  390. void CMainFrame::OnOpenFromTray() 
  391. {
  392.     this->ShowWindow(SW_RESTORE);
  393.    HideIcon();
  394.    ShowTaskBarButton(TRUE);
  395. }
  396.  
  397. void CMainFrame::OnShowWindow(BOOL bShow, UINT nStatus) 
  398. {
  399.     CMDIFrameWnd::OnShowWindow(bShow, nStatus);
  400.     
  401.     // I don't know what the following code needs to be here for,
  402.     // but it makes a massive headache on Win 9x (stack overflow).
  403. /*    
  404.     if (bShow)
  405.     {
  406.         HideIcon();
  407.         ShowTaskBarButton(TRUE);
  408.     }
  409. */
  410. }
  411.  
  412. /////////////////////////////////////////////////////////////////////////////
  413. // Tray Icon manipulation
  414.  
  415. void CMainFrame::MoveToRight()
  416. {
  417.     HideIcon();
  418.     ShowIcon();
  419. }
  420.  
  421. void CMainFrame::RemoveIcon()
  422. {
  423.     if (!m_bEnabled) return;
  424.  
  425.     m_tnd.uFlags = 0;
  426.     Shell_NotifyIcon(NIM_DELETE, &m_tnd);
  427.     m_bEnabled = FALSE;
  428. }
  429.  
  430. void CMainFrame::HideIcon()
  431. {
  432.     if (m_bEnabled && !m_bHidden) {
  433.         m_tnd.uFlags = NIF_ICON;
  434.         Shell_NotifyIcon (NIM_DELETE, &m_tnd);
  435.         m_bHidden = TRUE;
  436.     }
  437. }
  438.  
  439. void CMainFrame::ShowIcon()
  440. {
  441.     if (m_bEnabled && m_bHidden) {
  442.         m_tnd.uFlags = NIF_MESSAGE | NIF_ICON | NIF_TIP;
  443.         Shell_NotifyIcon(NIM_ADD, &m_tnd);
  444.         m_bHidden = FALSE;
  445.     }
  446. }
  447.  
  448. BOOL CMainFrame::SetIcon(HICON hIcon)
  449. {
  450.     if (!m_bEnabled) return FALSE;
  451.  
  452.     m_tnd.uFlags = NIF_ICON;
  453.     m_tnd.hIcon = hIcon;
  454.  
  455.     return Shell_NotifyIcon(NIM_MODIFY, &m_tnd);
  456. }
  457.  
  458. BOOL CMainFrame::SetIcon(LPCTSTR lpszIconName)
  459. {
  460.     HICON hIcon = AfxGetApp()->LoadIcon(lpszIconName);
  461.  
  462.     return SetIcon(hIcon);
  463. }
  464.  
  465. BOOL CMainFrame::SetIcon(UINT nIDResource)
  466. {
  467.     HICON hIcon = AfxGetApp()->LoadIcon(nIDResource);
  468.  
  469.     return SetIcon(hIcon);
  470. }
  471.  
  472. BOOL CMainFrame::SetStandardIcon(LPCTSTR lpIconName)
  473. {
  474.     HICON hIcon = LoadIcon(NULL, lpIconName);
  475.  
  476.     return SetIcon(hIcon);
  477. }
  478.  
  479. BOOL CMainFrame::SetStandardIcon(UINT nIDResource)
  480. {
  481.     HICON hIcon = LoadIcon(NULL, MAKEINTRESOURCE(nIDResource));
  482.  
  483.     return SetIcon(hIcon);
  484. }
  485.  
  486. HICON CMainFrame::GetIcon() const
  487. {
  488.     return (m_bEnabled)? m_tnd.hIcon : NULL;
  489. }
  490.  
  491. /////////////////////////////////////////////////////////////////////////////
  492. // Tray Icon tooltip text manipulation
  493.  
  494. BOOL CMainFrame::SetTooltipText(LPCTSTR pszTip)
  495. {
  496.     if (!m_bEnabled) return FALSE;
  497.  
  498.     m_tnd.uFlags = NIF_TIP;
  499.     _tcscpy(m_tnd.szTip, pszTip);
  500.  
  501.     return Shell_NotifyIcon(NIM_MODIFY, &m_tnd);
  502. }
  503.  
  504. BOOL CMainFrame::SetTooltipText(UINT nID)
  505. {
  506.     CString strText;
  507.     VERIFY(strText.LoadString(nID));
  508.  
  509.     return SetTooltipText(strText);
  510. }
  511.  
  512. CString CMainFrame::GetTooltipText() const
  513. {
  514.     CString strText;
  515.     if (m_bEnabled)
  516.         strText = m_tnd.szTip;
  517.  
  518.     return strText;
  519. }
  520.  
  521. /////////////////////////////////////////////////////////////////////////////
  522. // Tray Icon notification window stuff
  523.  
  524. BOOL CMainFrame::SetNotificationWnd(CWnd* pWnd)
  525. {
  526.     if (!m_bEnabled) return FALSE;
  527.  
  528.     // Make sure Notification window is valid
  529.     ASSERT(pWnd && ::IsWindow(pWnd->GetSafeHwnd()));
  530.  
  531.     m_tnd.hWnd = pWnd->GetSafeHwnd();
  532.     m_tnd.uFlags = 0;
  533.  
  534.     return Shell_NotifyIcon(NIM_MODIFY, &m_tnd);
  535. }
  536.  
  537. CWnd* CMainFrame::GetNotificationWnd() const
  538. {
  539.     return CWnd::FromHandle(m_tnd.hWnd);
  540. }
  541.  
  542. /////////////////////////////////////////////////////////////////////////////
  543. // Tray Icon menu manipulation
  544.  
  545. BOOL CMainFrame::SetMenuDefaultItem(UINT uItem, BOOL bByPos)
  546. {
  547.     if ((m_DefaultMenuItemID == uItem) && (m_DefaultMenuItemByPos == bByPos)) 
  548.         return TRUE;
  549.  
  550.     m_DefaultMenuItemID = uItem;
  551.     m_DefaultMenuItemByPos = bByPos;   
  552.  
  553.     CMenu menu, *pSubMenu;
  554.  
  555.     if (!menu.LoadMenu(m_tnd.uID)) return FALSE;
  556.     if (!(pSubMenu = menu.GetSubMenu(0))) return FALSE;
  557.  
  558.     ::SetMenuDefaultItem(pSubMenu->m_hMenu, m_DefaultMenuItemID, m_DefaultMenuItemByPos);
  559.  
  560.     return TRUE;
  561. }
  562.  
  563. void CMainFrame::GetMenuDefaultItem(UINT& uItem, BOOL& bByPos)
  564. {
  565.     uItem = m_DefaultMenuItemID;
  566.     bByPos = m_DefaultMenuItemByPos;
  567. }
  568.  
  569. LRESULT CMainFrame::OnTrayNotification(UINT wParam, LONG lParam) 
  570. {
  571.     //Return quickly if its not for this tray icon
  572.     if (wParam != m_tnd.uID)
  573.         return 0L;
  574.  
  575.     CMenu menu, *pSubMenu;
  576.     CWnd* pTarget = AfxGetMainWnd();
  577.  
  578.     // Clicking with right button brings up a context menu
  579.     if (LOWORD(lParam) == WM_RBUTTONUP)
  580.     {    
  581.         if (!menu.LoadMenu(m_tnd.uID)) return 0;
  582.         if (!(pSubMenu = menu.GetSubMenu(0))) return 0;
  583.  
  584.         // Make chosen menu item the default (bold font)
  585.         ::SetMenuDefaultItem(pSubMenu->m_hMenu, m_DefaultMenuItemID, m_DefaultMenuItemByPos);
  586.  
  587.         // Display and track the popup menu
  588.         CPoint pos;
  589.         GetCursorPos(&pos);
  590.  
  591.         pTarget->SetForegroundWindow();  
  592.         ::TrackPopupMenu(pSubMenu->m_hMenu, 0, pos.x, pos.y, 0, 
  593.                          pTarget->GetSafeHwnd(), NULL);
  594.  
  595.         // BUGFIX: See "PRB: Menus for Notification Icons Don't Work Correctly"
  596.         pTarget->PostMessage(WM_NULL, 0, 0);
  597.  
  598.         menu.DestroyMenu();
  599.     } 
  600.     else if (LOWORD(lParam) == WM_LBUTTONDBLCLK) 
  601.     {
  602.         // double click received, the default action is to execute default menu item
  603.         pTarget->SetForegroundWindow();  
  604.  
  605.         UINT uItem;
  606.         if (m_DefaultMenuItemByPos)
  607.         {
  608.             if (!menu.LoadMenu(m_tnd.uID)) return 0;
  609.             if (!(pSubMenu = menu.GetSubMenu(0))) return 0;
  610.             uItem = pSubMenu->GetMenuItemID(m_DefaultMenuItemID);
  611.         }
  612.         else
  613.             uItem = m_DefaultMenuItemID;
  614.         
  615.         pTarget->SendMessage(WM_COMMAND, uItem, 0);
  616.  
  617.         menu.DestroyMenu();
  618.     }
  619.  
  620.     return 1;
  621. }
  622.  
  623. LRESULT CMainFrame::WindowProc(UINT message, WPARAM wParam, LPARAM lParam) 
  624. {
  625.     if (message == m_tnd.uCallbackMessage)
  626.         return OnTrayNotification(wParam, lParam);
  627.     
  628.     return CWnd::WindowProc(message, wParam, lParam);
  629. }
  630.  
  631.  
  632.  
  633. BOOL CMainFrame::OnNcActivate(BOOL bActive) 
  634. {
  635.     // Add iconize button to caption
  636.     DrawButtons( GetSafeHwnd() );
  637.  
  638.     return CMDIFrameWnd::OnNcActivate(bActive);
  639. }
  640.  
  641.  
  642. void CMainFrame::OnNcLButtonUp(UINT nHitTest, CPoint point) 
  643. {
  644.     HDC        hDc        ;
  645.     RECT    rPos    ;
  646.     RECT    rWin    ;
  647.  
  648.     // Redraw caption buttons
  649.     if ( ( hDc = ::GetWindowDC( GetSafeHwnd() ) ) != NULL )
  650.     {
  651.         // Convert mouse postion relative to caption rectangle
  652.         GetWindowRect(&rWin);
  653.         
  654.         // Calculate rectangle position of iconize button
  655.         CalcCaptionRect( GetSafeHwnd(), rPos );
  656.         rPos.top    += 2;    
  657.         rPos.bottom -= 2;
  658.         rPos.right  -= ( ( GetSystemMetrics( SM_CXSIZE ) * 3 ) );
  659.         rPos.left    = rPos.right - ( GetSystemMetrics( SM_CXSIZE ) - 2 );
  660.  
  661.         point.x = point.x - rWin.left;
  662.         point.y = point.y - rWin.top;
  663.  
  664.         // Check if mouse position is in rectangle
  665.         if ( PtInRect( &rPos, point) )
  666.         {    
  667.             SendMessage( WM_SYSCOMMAND, SC_MINIMIZE, 0 );
  668.             ShowWindow( SW_HIDE );
  669.             ShowIcon();
  670.             ShowTaskBarButton(FALSE);
  671.         }
  672.         else
  673.             CMDIFrameWnd::OnNcLButtonUp(nHitTest, point);
  674.  
  675.         DrawFrameControl( hDc, &rPos, DFC_BUTTON, DFCS_BUTTONPUSH );
  676.         DrawIconize( hDc, rPos.left, rPos.top, 0 );        
  677.         m_LBtnDown = FALSE;
  678.         DeleteObject( hDc );
  679.     }
  680. }
  681.  
  682. void CMainFrame::OnNcLButtonDown(UINT nHitTest, CPoint point) 
  683. {
  684.     HDC        hDc        ;
  685.     RECT    rPos    ;
  686.     RECT    rWin    ;
  687.  
  688.     // Redraw caption buttons
  689.     if ( ( hDc = ::GetWindowDC( GetSafeHwnd()) ) != NULL )
  690.     {
  691.         // Convert mouse postion relative to caption rectangle
  692.         ::GetWindowRect( GetSafeHwnd(), &rWin );
  693.         
  694.         // Calculate rectangle position of iconize button
  695.         CalcCaptionRect( GetSafeHwnd(), rPos );
  696.         rPos.top    += 2;    
  697.         rPos.bottom -= 2;
  698.         rPos.right  -= ( ( GetSystemMetrics( SM_CXSIZE ) * 3 ) );
  699.         rPos.left    = rPos.right - ( GetSystemMetrics( SM_CXSIZE ) - 2 );
  700.  
  701.         point.x = point.x - rWin.left;
  702.         point.y = point.y - rWin.top;
  703.  
  704.         // Check if mouse position is in rectangle
  705.         if ( PtInRect( &rPos, point) )
  706.         {
  707.             m_LBtnDown = TRUE;
  708.             DrawFrameControl( hDc, &rPos, DFC_BUTTON, DFCS_BUTTONPUSH | DFCS_PUSHED );
  709.             DrawIconize( hDc, rPos.left, rPos.top, 1 );            
  710.         }
  711.         else
  712.             CMDIFrameWnd::OnNcLButtonDown(nHitTest, point) ;
  713.         
  714.         DeleteObject( hDc );
  715.     }
  716. }
  717.  
  718. void CMainFrame::OnNcMouseMove(UINT nHitTest, CPoint point) 
  719. {
  720.     HDC        hDc        ;
  721.     RECT    rPos    ;
  722.     RECT    rWin    ;
  723.  
  724.     // Redraw caption buttons
  725.     if ( ( hDc = ::GetWindowDC( GetSafeHwnd() ) ) != NULL )
  726.     {
  727.         // Convert mouse postion relative to caption rectangle
  728.         ::GetWindowRect( GetSafeHwnd(), &rWin );
  729.         
  730.         // Calculate rectangle position of iconize button
  731.         CalcCaptionRect( GetSafeHwnd(), rPos );
  732.         rPos.top    += 2;    
  733.         rPos.bottom -= 2;
  734.         rPos.right  -= ( ( GetSystemMetrics( SM_CXSIZE ) * 3 ) );
  735.         rPos.left    = rPos.right - ( GetSystemMetrics( SM_CXSIZE ) - 2 );
  736.  
  737.         point.x = point.x - rWin.left;
  738.         point.y = point.y - rWin.top;
  739.  
  740.         // Check if mouse position is in rectangle
  741.         if ( PtInRect( &rPos, point) )
  742.         {
  743.             DrawFrameControl( hDc, &rPos, DFC_BUTTON, ( m_LBtnDown ) ? DFCS_BUTTONPUSH | DFCS_PUSHED : DFCS_BUTTONPUSH );
  744.             DrawIconize( hDc, rPos.left, rPos.top, ( m_LBtnDown ) ? 1 : 0 );
  745.         }
  746.         else
  747.         {
  748.             m_LBtnDown = FALSE;
  749.             DrawFrameControl( hDc, &rPos, DFC_BUTTON, DFCS_BUTTONPUSH );
  750.             DrawIconize( hDc, rPos.left, rPos.top, 0 );
  751.             CMDIFrameWnd::OnNcMouseMove(nHitTest, point) ;
  752.         }
  753.     
  754.         DeleteObject( hDc );
  755.     }
  756. }
  757.  
  758. void CMainFrame::OnNcPaint() 
  759. {
  760.     CMDIFrameWnd::OnNcPaint();
  761.  
  762.     // Add iconize button to caption
  763.     DrawButtons( GetSafeHwnd() );
  764. }
  765.  
  766.  
  767. void CMainFrame::CalcCaptionRect( HWND hWnd, RECT& Rect )
  768. {    
  769.     DWORD dStyle        ;
  770.     SIZE  sFrame        ;
  771.     int      Icon            ;    
  772.  
  773.     // Get frame size of window
  774.     dStyle    = GetWindowLong( hWnd, GWL_STYLE );
  775.     sFrame.cx = GetSystemMetrics( ( dStyle & WS_THICKFRAME ) ? SM_CXSIZEFRAME : SM_CXFIXEDFRAME );
  776.     sFrame.cy = GetSystemMetrics( ( dStyle & WS_THICKFRAME ) ? SM_CYSIZEFRAME : SM_CYFIXEDFRAME );
  777.  
  778.     // Get width of icon/button in caption
  779.     Icon = GetSystemMetrics( SM_CXSIZE );
  780.  
  781.     // Calculate rectangle dimensions
  782.     ::GetWindowRect( hWnd, &Rect );
  783.     Rect.bottom -= Rect.top;
  784.     Rect.right  -= Rect.left;
  785.     Rect.top     = 0;
  786.     Rect.left    = 0;
  787.  
  788.     Rect.left   += sFrame.cx;
  789.     Rect.right  -= sFrame.cx;
  790.     Rect.top    += sFrame.cy;
  791.     Rect.bottom  = Rect.top + GetSystemMetrics( SM_CYCAPTION )
  792.                             - GetSystemMetrics( SM_CYBORDER );
  793. }
  794.  
  795.  
  796.  
  797. void CMainFrame::DrawIconize( HDC hDc, int x, int y, int off )
  798. {
  799.     //
  800.     // TODO:
  801.     // This should be done flikker free by means
  802.     // of a memory DC
  803.     //
  804.  
  805.     HPEN Pen, oldPen;
  806.     HBRUSH Brush, oldBrush;
  807.  
  808.     int     cxBtn = GetSystemMetrics( SM_CXSIZE ) - 5;
  809.     int  cyBtn = GetSystemMetrics( SM_CYSIZE ) - 5;
  810.     
  811.     Pen = CreatePen( PS_SOLID, 1, RGB( 0, 0, 0 ) );
  812.     oldPen = (HPEN)   SelectObject( hDc, Pen );
  813.  
  814.     Brush = CreateSolidBrush( RGB(0,0,0) );
  815.     oldBrush = (HBRUSH) SelectObject(hDc, Brush);
  816.  
  817.     // Use a filled circle in the bottom right corner to signify minimize to icon.
  818.     Ellipse(hDc, x + 1 + cxBtn*1/2 + off, y + 1 + cyBtn*1/2,
  819.         x + 1 + cxBtn*5/6, y + 1 + cyBtn*3/4);
  820.  
  821.     SelectObject( hDc, oldPen );
  822.     DeleteObject( Pen );
  823.  
  824.     SelectObject( hDc, oldBrush );
  825.     DeleteObject( Brush );
  826. }
  827.  
  828. void CMainFrame::DrawButtons( HWND hWnd )
  829. {
  830.     HDC      hDc        ;
  831.     RECT  rCap        ;
  832.     DWORD dStyle    ;
  833.     DWORD dExStyle    ;
  834.     int      cxBtn        ;
  835.     int   cyBtn        ;
  836.     RECT  rPos        ;
  837.  
  838.     // Get window device context        
  839.     if ( ( hDc = ::GetWindowDC( hWnd ) ) != NULL )
  840.     {
  841.         // Get caption coordinates
  842.         CalcCaptionRect( hWnd, rCap );
  843.  
  844.         // Get window style
  845.         dStyle   = GetWindowLong( hWnd, GWL_STYLE );
  846.         dExStyle = GetWindowLong( hWnd, GWL_EXSTYLE );
  847.     
  848.         // Check if we have a caption
  849.         if ( ( dStyle & WS_CAPTION ) == WS_CAPTION )
  850.         {
  851.             // Get button dimensions
  852.             cxBtn = GetSystemMetrics( SM_CXSIZE );
  853.             cyBtn = GetSystemMetrics( SM_CYSIZE );
  854.  
  855.             // Calc position and draw close button
  856.             rPos.top    = rCap.top + 2;
  857.             rPos.bottom = rCap.bottom - 2;
  858.             rPos.right  = rCap.right - 2;
  859.             rPos.left   = rCap.right - 2 - ( cxBtn - 2);
  860.             DrawFrameControl( hDc, &rPos, DFC_CAPTION, DFCS_CAPTIONCLOSE );
  861.  
  862.             // Calc position and draw maximize<->restore/help button
  863.             if ( ( dStyle & WS_MAXIMIZEBOX ) == WS_MAXIMIZEBOX )
  864.             {
  865.                 rPos.right -= cxBtn;
  866.                 rPos.left  -= cxBtn;
  867.                 DrawFrameControl( hDc, &rPos, DFC_CAPTION, ::IsZoomed( hWnd ) ? DFCS_CAPTIONRESTORE 
  868.                                                                             : DFCS_CAPTIONMAX );
  869.             }
  870.             else if ( ( dExStyle & WS_EX_CONTEXTHELP ) == WS_EX_CONTEXTHELP )
  871.             {
  872.                 rPos.right -= cxBtn;
  873.                 rPos.left  -= cxBtn;
  874.                 DrawFrameControl( hDc, &rPos, DFC_CAPTION, DFCS_CAPTIONHELP );
  875.             }
  876.  
  877.             // Calc position and draw minimize and iconize button
  878.             if ( ( dStyle & WS_MINIMIZEBOX ) == WS_MINIMIZEBOX )
  879.             {
  880.                 // Minimize button
  881.                 rPos.right -= ( cxBtn - 2 );
  882.                 rPos.left  -= ( cxBtn - 2 );
  883.                 DrawFrameControl( hDc, &rPos, DFC_CAPTION, DFCS_CAPTIONMIN );
  884.  
  885.                 // Iconize button
  886.                 rPos.right -= ( cxBtn - 2 ) + 2;
  887.                 rPos.left  -= ( cxBtn - 2 ) + 2;
  888.                 DrawFrameControl( hDc, &rPos, DFC_BUTTON, DFCS_BUTTONPUSH );
  889.  
  890.                 // Draw a btnface
  891.                 DrawIconize( hDc, rPos.left, rPos.top, 0 );
  892.             }
  893.         }
  894.  
  895.         // Release device context
  896.         ::ReleaseDC( hWnd, hDc );
  897.     }
  898. }
  899.  
  900. void CMainFrame::OnPaint() 
  901. {
  902.     CPaintDC dc(this); // device context for painting
  903.     
  904.     // Add iconize button to caption
  905.     DrawButtons( GetSafeHwnd() );
  906. }
  907.